package com.tanhua.sso.service;

import com.alibaba.dubbo.config.annotation.Reference;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;

import com.tanhua.common.mapper.UserMapper;
import com.tanhua.common.pojo.User;
import com.tanhua.dubbo.server.api.HuanXinApi;
import io.jsonwebtoken.ExpiredJwtException;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.digest.DigestUtils;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

/**
 * 陈功炳
 * @since 2021/11/23 18:39
 */
@Service
@Slf4j
public class UserService {
    @Autowired
    private RedisTemplate<String,String> redisTemplate;

    @Autowired
    private UserMapper userMapper;

    @Value("${jwt.secret}")
    private String secret;

    @Reference(version = "1.0.0")
    private HuanXinApi huanXinApi;

    /**
     * 根据登录用户id生成token
     * @param phone 用户手机号
     * @param code 生成的验证码
     * @return 返回的数据
     */
    public Map<String, Object> login(String phone, String code) {
        //设置redis
        String rediskey = "CHECK_CODE_"+phone;
        boolean isNew = false;
        //校验验证码
        String redisData = redisTemplate.opsForValue().get(rediskey);
        if (!code.equals(redisData)){
            return null;//验证码错误
        }
        //验证码校验完后需要删除
       redisTemplate.delete(rediskey);


        QueryWrapper<User> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("mobile",phone);

        User user = userMapper.selectOne(queryWrapper);

        if (null == user){
            //需要注册用户
            user = new User();
            user.setMobile(phone);
            user.setPassword(DigestUtils.md2Hex("123456"));//DigestUtils.md2Hex加密

            //注册新用户
            userMapper.insert(user);
            isNew = true;

            //注册环信用户
            Boolean result = this.huanXinApi.register(user.getId());
            if (!result) {
                //注册环信失败，记录日志
                log.error("注册环信用户失败~ userId = " + user.getId());
            }
        }

        //生成token
        Map<String,Object> claims = new HashMap<>();
        claims.put("id",user.getId());
        // 生成token
        String token = Jwts.builder()
                .setClaims(claims) //payload，存放数据的位置，不能放置敏感数据，如：密码等
                .signWith(SignatureAlgorithm.HS256, secret) //设置加密方法和加密盐
                .setExpiration(new DateTime().plusHours(12).toDate()) //设置过期时间，12小时后过期
                .compact();
        Map<String,Object> map = new HashMap<>();
        map.put("token",token);
        map.put("isNew",isNew);
        return map;
    }

    /**
     * 校验token 查询用户id
     * @author 陈功炳
     * @since 2021/11/23 18:39
     */
    public User queryUserByToKen(String token) {
        User user = new User();
        try {
            // 通过token解析数据
            Map<String, Object> body = Jwts.parser()
                    .setSigningKey(secret)
                    .parseClaimsJws(token)
                    .getBody();
             //{mobile=1333333333, id=2, exp=1605513392}


            user.setId(Long.valueOf(body.get("id").toString()));
            //需要返回user对象中的mobile,需要查询数据库获取到mobile数据
            //如果每次都要查询数据库，必然会导致性能问题，需要对用户的手机号进行缓存操作
            //数据缓存时，需要设置过期时间，过期时间要与token的时间一致
            String redisKey = "TANHUA_USER_MOBILE_"+user.getId();
            if (redisTemplate.hasKey(redisKey)){
                String mobile = redisTemplate.opsForValue().get(redisKey);
                user.setMobile(mobile);
            }else {
                //查询数据库
                User u = userMapper.selectById(user.getId());
                user.setMobile(u.getMobile());
                //将手机号写入redis中
                //在jwt中的过期时间的单位为:秒
                long timeout = Long.valueOf(body.get("exp").toString()) * 1000 - System.currentTimeMillis();
                redisTemplate.opsForValue().set(redisKey,u.getMobile(),timeout, TimeUnit.MILLISECONDS);
            }

        } catch (ExpiredJwtException e) {
            log.info("token已经过期! token = "+ token);
            System.out.println("token已经过期！");
        } catch (Exception e) {
            log.error("token不合法! token = "+ token,e);
        }
        return user;
    }

}
